import time
# 导入selenium
from selenium import webdriver
from selenium.webdriver.common.by import By
from selenium.webdriver.chrome.options import Options
# from selenium.webdriver.chrome.service import Service


# 设置 Chrome 选项
chrome_options = Options()
chrome_options.add_argument("--headless")  # 启用无头模式
chrome_options.add_argument("--disable-gpu")  # 禁用 GPU 加速（某些系统需要）
chrome_options.add_argument("--no-sandbox")  # 禁用沙盒模式
chrome_options.add_argument("--disable-dev-shm-usage")  # 解决内存不足问题


# 打开chrome浏览器
browser = webdriver.Chrome(options=chrome_options)

# 打开指定的页面
browser.get("https://wap.jczq.com/#/index")

# 等待页面加载完
# browser.implicitly_wait(5)
time.sleep(2)

# 获取页面源码
html = browser.page_source
# 写入文件
file = open('./runtime/html1.html', 'w', encoding='utf-8')
file.write(html)
file.close()
#print(html)

# 页面可能会有弹窗，先关闭所有弹窗
toasts = browser.find_elements(By.XPATH, '//*[@id="app"]/div[1]/div[2]/div/img[2]')
for toast in toasts:
    toast.click()

# 模拟点击tab【高命中率】
highElement = browser.find_element(By.XPATH, '//*[@id="app"]/div[1]/div/div/div/div[2]/div[2]/div/div[1]/div/div[2]')
print(highElement.text)
highElement.click()

# 获取页面源码
# 等待页面加载完
time.sleep(2)
html2 = browser.page_source
# 写入文件
file2 = open('./runtime/html2.html', 'w', encoding='utf-8')
file2.write(html2)
file2.close()

# 设置五秒后执行下一步
time.sleep(10)
# 关闭浏览器
browser.quit()